package _11_整理题目._4_动态规划._打家劫舍;

import org.junit.Test;


/**
 * https://leetcode.cn/problems/delete-and-earn/
 * 给你一个整数数组nums，你可以对它进行一些操作。
 * 每次操作中，选择任意一个nums[i]，删除它并获得nums[i]的点数。之后，
 * 你必须删除 所有 等于nums[i] - 1 和 nums[i] + 1的元素。
 * 开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
 *
 * 示例2：
 * 输入：nums = [2,2,3,3,3,4]
 * 输出：9
 * 解释：
 * 删除 3 获得 3 个点数，接着要删除两个 2 和 4 。
 * 之后，再次删除 3 获得 3 个点数，再次删除 3 获得 3 个点数。
 * 总共获得 9 个点数。
 *
 *
 */
public class _03_删除并获得点数 {

    public int deleteAndEarn(int[] nums) {
        int maxValue = Integer.MIN_VALUE;
        for(int num : nums) {
            maxValue = Math.max(maxValue, num);
        }
        int[] sum = new int[maxValue+1];
        for(int num : nums) {
            sum[num] += num;
        }
        int ans = rob(sum);
        return ans;
    }

    public int rob(int[] nums) {
        int n = nums.length;
        if(n==1) return nums[0];
        if(n==2) return Math.max(nums[0], nums[1]);

        int a = nums[0];
        int b = Math.max(nums[0], nums[1]);
        int c = 0;
        for(int i=2; i<n; i++) {
            c = Math.max(b, a+nums[i]);
            a = b;
            b = c;
        }
        return c;
    }

    @Test
    public void main() {
        System.out.println();
    }
}
